import pandas as pd

from plot_line import plot_line

file1 = '滇中各城市-新型城镇化与生态环境-协调度a2b2.xlsx'
title1 = '滇中各城市-新型城镇化与生态环境-协调度演变趋势-类型1(a=1/2,b=1/2)'
df1 = pd.read_excel(file1, index_col=0)
file2 = '滇中各城市-新型城镇化与生态环境-协调度a1b2.xlsx'
title2 = '滇中各城市-新型城镇化与生态环境-协调度演变趋势-类型2(a=1/3,b=2/3)'
df2 = pd.read_excel(file2, index_col=0)
file3 = '滇中各城市-新型城镇化与生态环境-协调度a2b1.xlsx'
title3 = '滇中各城市-新型城镇化与生态环境-协调度演变趋势-类型3(a=2/3,b=1/3)'
df3 = pd.read_excel(file3, index_col=0)

# 昆明市，曲靖市，玉溪市，楚雄州，红河州
city_name = '昆明市'
new_df = pd.DataFrame()
new_df['类型1(a=1/2,b=1/2)'] = df1[city_name]
new_df['类型2(a=1/3,b=2/3)'] = df2[city_name]
new_df['类型3(a=2/3,b=1/3)'] = df3[city_name]
print('-----------new_df-----------')
print(new_df)
title = city_name + '-新型城镇化与生态环境-不同类型协调度演变趋势'
print(title)
plot_line(new_df, title)
